#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<fcntl.h>
#include<errno.h>//引入错误值errno
#define BUF_SIZE 2//把缓冲大小改小，以验证边缘触发
#define EPOLL_SIZE 50
//epoll的边缘触发（ET）方式工作
//边缘触发方式中，输入缓冲收到数据时仅注册1次该事件
//必须使用非阻塞read/write，通过fcntl修改
void setnonblockingmode(int fd);//修改成非阻塞IO，避免因阻塞方式工作的read&write函数引起服务器端的长时间停顿
void error_handling(char *buf);
int main(int argc,char *argv[])
{
	int serv_sock,clnt_sock;
	struct sockaddr_in serv_adr,clnt_adr;
	socklen_t adr_sz;
	int str_len,i;
	char buf[BUF_SIZE];
	//epoll方式通过结构体epoll_event将发生的文件描述符单独集中到一起
	struct epoll_event *ep_events;
	struct epoll_event event;
	//epoll文件描述符
	int epfd,event_cnt;
	
	if(argc!=2)
	{
		printf("Usage:%s<port>\n",argv[0]);
		exit(1);
	}
	serv_sock=socket(PF_INET,SOCK_STREAM,0);
	memset(&serv_adr,0,sizeof(serv_adr));
	serv_adr.sin_family=AF_INET;
	serv_adr.sin_addr.s_addr=htonl(INADDR_ANY);
	serv_adr.sin_port=htons(atoi(argv[1]));
	
	if(bind(serv_sock,(struct sockaddr*)&serv_adr,sizeof(serv_adr))==-1)
		error_handling("bind() error");
	if(listen(serv_sock,5)==-1)
		error_handling("listen() error");
	
	//创建epoll例程，返回的是文件描述符，主要用于区分epoll例程
	epfd=epoll_create(EPOLL_SIZE);
	ep_events=malloc(sizeof(struct epoll_event)*EPOLL_SIZE);//需要动态分配
	
	setnonblockingmode(serv_sock);//设置非阻塞IO
	//epoll_event结构体用于保存发生事件的文件描述符集合
	//用于注册关注的事件
	event.events=EPOLLIN;//发生需要读取数据的情况（事件）时
	event.data.fd=serv_sock;//监控的文件描述符
	//在其内部注册监视对象文件描述符
	//int epoll_ctl(int epfd,int op,int fd,struct epoll_event *event)
	//参数op主要是例程对监视对象的操作，添加，删除或者更改
	//event是某一个例程所要监听的事件
	epoll_ctl(epfd,EPOLL_CTL_ADD,serv_sock,&event);//这里就是添加一个serv_sock
	//而ep_events是100个例程所发生的事件的结构体的首地址
	
	while(1)
	{
		//int epoll_wait(int epfd,struct epoll_event *events,int maxevents,int timeout)
		//等待事件发生，当事件发生时返回发生事件的文件描述符数
		//返回的是满足事件发生的ep_events，没发生的不用返回
		event_cnt=epoll_wait(epfd,ep_events,EPOLL_SIZE,-1);
		if(event_cnt==-1)//失败时
		{
			puts("epoll_wait() error");
			break;
		}
		puts("return epoll_wait");//输入缓冲有数据就会通知事件，所以epoll_wait会一直调用
		//所以此时遍历的是发生了事件的ep_events
		for(i=0;i<event_cnt;i++)
		{
			if(ep_events[i].data.fd==serv_sock)//发生连接请求
			{
				adr_sz=sizeof(clnt_adr);
				clnt_sock=accept(serv_sock,(struct sockaddr*)&clnt_adr,&adr_sz);
				setnonblockingmode(clnt_sock);//设置非阻塞IO
				event.events=EPOLLIN|EPOLLET;//EPOLLET设置边缘触发
				event.data.fd=clnt_sock;
				epoll_ctl(epfd,EPOLL_CTL_ADD,clnt_sock,&event);
				printf("connected client:%d\n",clnt_sock);
			}
			else//读操作
			{
				//由于是边缘触发，只触发1次时间，故需要一次性把输入缓冲中的数据读出来
				while(1)
				{
					str_len=read(ep_events[i].data.fd,buf,BUF_SIZE);
					if(str_len==0)//关闭请求,客户端发来EOF时
					{
						//从例程epfd删除此文件描述符，不再监听
						epoll_ctl(epfd,EPOLL_CTL_DEL,ep_events[i].data.fd,NULL);
						close(ep_events[i].data.fd);//关闭客户端套接字
						printf("closed client:%d\n",ep_events[i].data.fd);
						break;
					}
					//read函数返回-1且errno值为EAGAIN时，意味着读取了输入缓冲中的全部数据，就跳出读的无限循环
					else if(str_len<0)
					{
						if(errno==EAGAIN)
							break;
					}
					else
					{
						write(ep_events[i].data.fd,buf,str_len);
					}
				}	
			}
		}
		
	}
	close(serv_sock);
	close(epfd);
	return 0;
}
void setnonblockingmode(int fd)
{
	//获取之前设置的属性信息
	int flag=fcntl(fd,F_GETFL,0);
	//添加非阻塞O_NONBLOCK标志
	fcntl(fd,F_SETFL,flag|O_NONBLOCK);
}
void error_handling(char *buf)
{
	fputs(buf,stderr);
	fputc('\n',stderr);
	exit(1);
}

